package com.ym.config;

import cn.dev33.satoken.reactor.filter.SaReactorFilter;
import cn.dev33.satoken.router.SaRouter;
import cn.dev33.satoken.stp.StpUtil;
import cn.dev33.satoken.util.SaResult;
import com.google.common.collect.ImmutableSet;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.Set;

/**
 * [Sa-Token 权限认证] 配置类
 * @author click33
 */
@Configuration
public class SaTokenConfigure {
    // 根据需求放行
    private static final Set<String> EXCLUDE = ImmutableSet.<String>builder()
            .add("/webjars/**")
            .add("/swagger-resources")
            .add("/doc.html")
            .add("/*/v3/api-docs")
            .build();

    // 注册 Sa-Token全局过滤器
    @Bean
    public SaReactorFilter getSaReactorFilter() {
        return new SaReactorFilter()
                // 拦截地址
                .addInclude("/**")    /* 拦截全部path */
                // 开放地址
                .addExclude(EXCLUDE.toArray(new String[0]))
                // 鉴权方法：每次访问进入
                .setAuth(obj -> {
                    // 登录校验 -- 拦截所有路由，并排除/auth/login 用于开放登录
                    SaRouter.match("/**", "/user/login", r -> StpUtil.checkLogin());

                    // 权限认证 -- 不同模块, 校验不同权限
                    // 如果是微服务则前面需要加上微服务名称，这里就省略不写了，这里可以实现对指定服务统一先鉴权，然后还可以在具体服务实现更深层次的鉴权
                    //SaRouter.match("/user/**", r -> StpUtil.checkPermission("user"));
                    //SaRouter.match("/admin/**", r -> StpUtil.checkPermission("admin"));
                    //SaRouter.match("/goods/**", r -> StpUtil.checkPermission("goods"));
                    //SaRouter.match("/orders/**", r -> StpUtil.checkPermission("orders"));

                    // 更多匹配 ...  */
                })
                // 异常处理方法：每次setAuth函数出现异常时进入
                .setError(e -> SaResult.error(e.getMessage()))
                ;
    }
}
